import torch

def save_model(model, path):
    """
    保存模型状态字典
    
    参数:
        model: 要保存的模型
        path: 保存路径
    """
    torch.save(model.state_dict(), path)

def load_model(model, path):
    """
    加载模型状态字典
    
    参数:
        model: 要加载状态的模型
        path: 模型文件路径
        
    返回:
        model: 加载了状态的模型
    """
    model.load_state_dict(torch.load(path))
    return model